import ai as ai
from openai import OpenAI

my_key = "自己的api"
url = "https://api.deepseek.com"

def add_user(user_name, prompt = None):
   ai.add_user(user_name, prompt)

def add_history(user_name, role, content):
   ai.add_history(user_name, role, content)

def chat(user, prompt):
   return ai.chat(user, prompt)

ai.change_client(
    OpenAI(
        api_key=my_key,
        base_url=url
    ),
    "deepseek-chat"
)